package leetcode.code;

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;

/**
 * 给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。
 *
 * 说明：
 *
 * 拆分时可以重复使用字典中的单词。
 * 你可以假设字典中没有重复的单词。
 * 示例 1：
 *
 * 输入: s = "leetcode", wordDict = ["leet", "code"]
 * 输出: true
 * 解释: 返回 true 因为 "leetcode" 可以被拆分成 "leet code"。
 * 示例 2：
 *
 * 输入: s = "applepenapple", wordDict = ["apple", "pen"]
 * 输出: true
 * 解释: 返回 true 因为 "applepenapple" 可以被拆分成 "apple pen apple"。
 *      注意你可以重复使用字典中的单词。
 * 示例 3：
 *
 * 输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
 * 输出: false
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/word-break
 */
public class 单词拆分 {

    /**
     * 动态规划
     */
    static class Solution {
        public boolean wordBreak(String s, List<String> wordDict) {
            int m = s.length();
            if (s.length() == 0) return false;
            boolean[] f = new boolean[m + 1];
            f[0] = true;
            for(int i = 1; i <= s.length(); i++){
                for(String w: wordDict){
                    int len = w.length();
                    if(i >= len && f[i - len] == true
                            && s.substring(i - len, i).equals(w)){
                        f[i] = true;
                        break;
                    }
                }

            }
            return f[m];
        }
    }

    /**
     * dfs复杂度过高
     */
    static class Solution_1 {

        private HashSet<String> wordMap;
        private int maxLength = 0;

        public boolean wordBreak(String s, List<String> wordDict) {
            wordMap = new HashSet<>();
            wordMap.addAll(wordDict);
            for (String i : wordDict){
                maxLength = Math.max(maxLength, i.length());
            }
            return dfs(s, 0);
        }

        public boolean dfs(String s, int start) {
            if (start == s.length()){
                return true;
            }else if (start > s.length()){
                return false;
            }else {
                int i = 1;
                while (start+i <= s.length() && i <= maxLength){
                    if (wordMap.contains(s.substring(start, start+i))){
                        if (dfs(s, start+i)) return true;
                    }
                    i++;
                }
                return false;
            }
        }
    }


    public static void main(String[] args){
        Solution s = new Solution();
        System.out.println(s.wordBreak("applepenapple", Arrays.asList(new String[]{"apple", "pen", "app"})));
    }
}
